package com.study4j.generator;

import java.io.IOException;
import java.util.Properties;

import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PropertiesLoaderUtils;

import lombok.Data;


/**   
 * @description  自定义配置信息读取
 * @author Jiangbin  
 * @date 2019年10月11日 上午11:51:49 
 * @version 1.0.0  
 */
@Data
public class CodeGeneConfig {
    private Properties props;

    public CodeGeneConfig() {
        Resource resource = new ClassPathResource("/code.properties");
        try {
            props = PropertiesLoaderUtils.loadProperties(resource);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取属性
     *
     * @param key
     * @return
     */
    public String getProperty(String key) {
        return props == null ? null : props.getProperty(key);
    }

    /**
     * 获取属性
     *
     * @param key
     * @param defaultValue
     * @return
     */
    public String getProperty(String key, String defaultValue) {
        return props == null ? null : props.getProperty(key, defaultValue);
    }

    /**
     * 获取属性
     *
     * @return
     */
    public Properties getProperties() {
        return props;
    }
}
